package com.fll.codeworld.doc;

import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.Version;
import lombok.extern.slf4j.Slf4j;

import java.io.*;
import java.util.Map;
import java.util.UUID;

/**
 * @author fll
 * @description Freemarker工具类
 * @date 2022/4/20 16:15
 */
@Slf4j
public class FreemarkerUtil {

    /**
     * 根据ftl模板生成word文件并保存到磁盘
     *
     * @param dataMap      word中需要展示的动态数据，用map集合来保存
     * @param templateName word模板名称，例如：word.ftl
     * @param ftlPath      模板存放路径   例如: D:/wordFtl/
     * @param savePath     生成文档的存放路径   例如: D:/word/
     */
    public static String createWord(Map dataMap, String templateName, String ftlPath, String savePath) {
        //文件存放路径
        String newWordPath = savePath + UUID.randomUUID() + ".docx";
        try {
            //创建配置实例
            Configuration configuration = initConfiguration(ftlPath);
            Template template = configuration.getTemplate(templateName);
            //输出文件  输出文档路径及名称
            File outFile = new File( newWordPath);
            if (!outFile.getParentFile().exists()) {
                outFile.getParentFile().mkdirs();
            }
            //将模板和数据模型合并生成文件
            Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(outFile), "UTF-8"));
            //生成文件
            template.process(dataMap, out);
            //关闭流
            out.flush();
            out.close();
        } catch (Exception e) {
            log.error("*****创建word发生异常*****");
            e.printStackTrace();
        }
        log.info("======创建word文件完成======");
        return newWordPath;
    }

    /**
     * 加载配置
     *
     * @param folderPath
     * @return
     */
    private static Configuration initConfiguration(String folderPath) {
        Configuration configuration = new Configuration(new Version("2.3.29"));
        configuration.setDefaultEncoding("utf-8");
        try {
            //这里对应模板的物理路径
            configuration.setDirectoryForTemplateLoading(new File(folderPath));
        } catch (IOException e) {
            log.error("初始化configuration失败，路径不存在：{}", folderPath);
            e.printStackTrace();
        }
        return configuration;
    }
}
